package com.wang.aliyun.controller;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.OSSObject;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;

/**
 * @author: wang
 * @date: 2021/7/30 21:00
 * @description:
 */
@Controller
public class FileDownloadController {

    @RequestMapping("download")
    public void download(@RequestParam("fileName") String fileName,
                         HttpServletResponse response
                         ) throws IOException {
        response.setContentType("application/octef-stream");
        // response.setHeader("Content-disposition", "attachment;fileName="+fileName);
        response.setHeader("Content-disposition", "inline;fileName="+fileName);


        // yourEndpoint填写Bucket所在地域对应的Endpoint。以华东1（杭州）为例，Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。
        String endpoint = "https://oss-cn-shenzhen.aliyuncs.com";
        // 阿里云账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM用户进行API访问或日常运维，请登录RAM控制台创建RAM用户。
        String accessKeyId = "LTAI5t6gP6eogVesu1Bjqcsv";
        String accessKeySecret = "mk7bJQBtxzAPWPSAOnm0SKMnGEog9S";
        // 填写Bucket名称。
        String bucketName = "wuqixiaowei";
        // 填写Object的完整路径。Object完整路径中不能包含Bucket名称。
        String objectName = fileName;

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        // ossObject包含文件所在的存储空间名称、文件名称、文件元信息以及一个输入流。
        OSSObject ossObject = ossClient.getObject(bucketName, objectName);
        BufferedInputStream bis = null;
        ServletOutputStream os = null;
        // 读取文件内容。
        try {
            bis = new BufferedInputStream(ossObject.getObjectContent());
            os = response.getOutputStream();
            byte[] bytes = new byte[2048];
            int len;
            while ( (len = bis.read(bytes)) != -1 ) {
                os.write(bytes,0,len);
            }
            os.flush();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (bis != null){
                try {
                    bis.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (os != null){
                try {
                    os.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        // 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。
        // 关闭OSSClient。
        ossClient.shutdown();
    }
}
